paddle.hub |
您所在的位置:网站首页 › paddle inference加载模型失败 › paddle.hub |
paddle.hub¶
paddle.hub 是预训练模型库的集合,用来复用社区生产力,方便加载发布在 github、gitee 以及本地的预训练模型。飞桨提供框架模型拓展相关的 API 以及支持的模型库列表。具体如下: 查看和加载 API 支持模型库列表 查看和加载 API¶API 名称 API 功能 list 查看 Repo 支持的模型列表 help 查看指定模型的文档 load 加载指定模型 支持模型列表¶模型名字 模型库 alexnet PaddleClas vgg11 PaddleClas vgg13 PaddleClas vgg16 PaddleClas vgg19 PaddleClas resnet18 PaddleClas resnet34 PaddleClas resnet50 PaddleClas resnet101 PaddleClas resnet152 PaddleClas squeezenet1_0 PaddleClas squeezenet1_1 PaddleClas densenet121 PaddleClas densenet161 PaddleClas densenet169 PaddleClas densenet201 PaddleClas densenet264 PaddleClas inceptionv3 PaddleClas inceptionv4 PaddleClas googlenet PaddleClas shufflenetv2_x0_25 PaddleClas mobilenetv1 PaddleClas mobilenetv1_x0_25 PaddleClas mobilenetv1_x0_5 PaddleClas mobilenetv1_x0_75 PaddleClas mobilenetv2_x0_25 PaddleClas mobilenetv2_x0_5 PaddleClas mobilenetv2_x0_75 PaddleClas mobilenetv2_x1_5 PaddleClas mobilenetv2_x2_0 PaddleClas mobilenetv3_large_x0_35 PaddleClas mobilenetv3_large_x0_5 PaddleClas mobilenetv3_large_x0_75 PaddleClas mobilenetv3_large_x1_0 PaddleClas mobilenetv3_large_x1_25 PaddleClas mobilenetv3_small_x0_35 PaddleClas mobilenetv3_small_x0_5 PaddleClas mobilenetv3_small_x0_75 PaddleClas mobilenetv3_small_x1_0 PaddleClas mobilenetv3_small_x1_25 PaddleClas resnext101_32x4d PaddleClas resnext101_64x4d PaddleClas resnext152_32x4d PaddleClas resnext152_64x4d PaddleClas resnext50_32x4d PaddleClas resnext50_64x4d PaddleClas bert PaddleNLP 代码示例¶ import paddle # PaddleClas models = paddle.hub.list('PaddlePaddle/PaddleClas:develop', source='github', force_reload=True,) print(models) docs = paddle.hub.help('PaddlePaddle/PaddleClas:develop', 'alexnet', source='github', force_reload=False,) print(docs) model = paddle.hub.load('PaddlePaddle/PaddleClas:develop', 'alexnet', source='github', force_reload=False, pretrained=True) data = paddle.rand((1, 3, 224, 224)) out = model(data) print(out.shape) # [1, 1000] # PaddleNLP docs = paddle.hub.help('PaddlePaddle/PaddleNLP:develop', model='bert',) print(docs) model, tokenizer = paddle.hub.load('PaddlePaddle/PaddleNLP:develop', model='bert', model_name_or_path='bert-base-cased') |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |